Improve printing of ASTs