Update virus_complete.py