Factored UI out of game and main