added functionality to registration screen