Simplify parsing of for loops