Fix UI bugs
Close #148 Logic for register button