slight edit to output