Create main.py

This is the file that need to be invoke when running the code:
python main.py