minor changes
Changed test cases