Update main.py

included main method