Added test
Changed output