clean up comments
add a test for pretty printer