Updated test cases for iteration 4

All test cases passed