Adding in login functionality - all tests passing