Update lab.py