Added test cases
Fixed IO output typo