Improved UX
Added functionality for register button.