Gui changes - fix register button