allow remove entity with id only
Add test for removing entities