Cleaning up the output