Changed Main.py to a class