Updates the main cpp file

Game states have comments depicting what code should be placed where.