def process_rooms(df):
    """Takes a dataframe and isolate the numbers of rooms.
    
    Parameters
    ----------
    df : 
        The dataframe to search.
 
    Returns
    -------
    The dataframe processed.
    """
    df['bedroom'] = df['bedroom'].str.split(' ').str[0]
    df['bathroom'] = df['bathroom'].str.split(' ').str[0]
    return df