Add code to register user