fixing test cases and also modifying ex according
fixed test cases