Update to python_lab.py that finishes the assignment and seems to upload correctly.