Added functionality to HUD