Adding in IEntity interface and made entity classes implement it