Created some repository methods and added references to the entities