Add Entity class