Removed unused import
Removing unecessary return
Added Epidemic pojo and refactored ZombieEpidemic to have no implementation details.