Redid the Game State Management

Found tutorial on how to implement game state management. Used they
tutorials implementation, changing it to fit my needs.