Added play state and transitions between play and menu states