Add comments in generated code for what blocks represent which tokens