Update main.py

Updated comments and objectives / to-do list.