Update main.py

outsourced the game class