Create lab_2.py