Update main.py

Initial comments on definition of variables was added