Fix minor bug
Fix up the pretty printer