[TASK] Modified Columns to add class row