Changed the way Zombie class checks entity types, and fixed Main.