Record more info in display-ast