def mecs_energy_fba_cleanup(fba, attr, **kwargs):
    """
    Clean up the EIA MECS energy FlowByActivity
    :param fba: df, FBA format
    :param attr: dictionary, attribute data from method yaml for activity set
    :param kwargs: optional, can also include bool 'download_FBAs_if_missing'
    :return: df, subset of EIA MECS Energy FBA
    """
    # subset the df to only include values where the unit = MJ
    fba = fba.loc[fba['Unit'] == 'MJ'].reset_index(drop=True)

    return fba