little clean up on the output