Done some minor corrections to the test cases