Add tests for pretty printer