Prefer enter() to init() for play state.

Also add ways to get back from game over screen