Updates to hud generation code