added a simple main.py to run the code