Add test cases
Updated output