Added test cases for part 6