Add check on login