Fix: support nested for loops.
Code style fixes.