Small edits to output