UI changes for registration