Finished State Pattern. Added game over state and restart function