Refactor example to use Long instead of Integer for primary keys