Add in handling for for loops.