Added support for more AST nodes:

- Array literals
- While loops
- For loops