Updated test cases

updated test cases for iteration 3