Adding HUD functionality