Enhance AST
Add for loops
Allow empty if and for bodies