add lab2.py