Update infections.py