added Week 4 and tried to copy code over to new main.py