added the main.py script