Adding JPA retrieve code