Improvements to registration screen