Added test for add_total_columns.