UI changes to register