Add table_columns method and test for it