Simplify for loop
Whoops