Update Lab2.py