Updating session4.py with current progress