Add support for pk aliases
Add get_columns utility function