made slight changes to test cases