Added @Entity to classes