def remove_census_not_in_ycom(census):
    """
    Function to remove data for Puerto Rico,
    which is not included in YCOM data.
    Input: dataframe of census data
    Output: dataframe of census data with Puerto
    Rico removed
    """
    census = census[census['State'] != 'Puerto Rico']
    return census