Update and rename lab2.py to lab3.py

framework for lab3