Update main for playing the game

TODO: create a separate method for playing instead of having it in main.