UI updated for the game also