Added doc to main.py