Update lab2.py