Restore outbreak.py