Cleanup + add file for prototyping AST code