Add a pass for code cleanup.

Add better printing of ParserRuleContexts.