need to clean up the output