Working on the registration screen