Add test for findByPK
Fix getColumns() to return correct PK