Added some check. Working in the HUD