added remove_columns to the migration