Create lab2.py