Added login/register back