Add with_join_columns method