[Refactor] Use nicer print callback function in IslAst