Fixing AST and print function. Only need to finish Graph in AST