def sort_data(data_df, sort):
    """Sort data in ascending or descending order

    Args:
        data_df (df): Dataframe of data to plot
        sort (str): 'ascending' or 'descending' order to sort in

    Returns:
        df: Dataframe with data sorted in the specified order
    """

    if sort == "ascending":
        data_df = data_df.sort_values(data_df.columns[0], ascending=True)
    elif sort == "descending":
        data_df = data_df.sort_values(data_df.columns[0], ascending=False)

    return data_df