need to clean up output