Slight modifications to login