add a function to retrieve primary key column