Add primary_key method to allow saving via activerecord