Improved the HUD