def rename(df, task):
    """Subset all columns starting with `task`
    and remove `task` from it
    """
    dfs = df.iloc[:, df.columns.str.match(task)]
    dfs.columns = dfs.columns.str.replace(task + " ", "")
    dfs['task'] = task
    return dfs