Update main.py

added a note where to continue when I continue working