slight update to output