Can now exit play state by pressing escape