Added test for output