Added support for the following statement types:

- ForStmt
- WhileStmt
- More assignment statements (including array set)