Adding main.py with Main class