Update to main.py

Adding Notes to self.