Added login/register code