Work on the game ui