Add lab2.py