Update Lab 2.py