added get_columns function