Adds more epidemy implementation