added comments the main.py