Added the Entity Class