Updating documentation
Cleaning up AST