Added removeEntity method