Add if for login/reg