Update lab3.py