Update lab2.py

WHY