Add test for login