Added main.py to project