Created main.py

These are my addition to the project