Added stuff for register and login