Updated lab4.py