minor changes to output