Integrating the UI into the game