Updated init.py with the MainMenu class.