No it was not difficult to figure out how to rewrite square using a for-loop. The strategy I used was; instead of repeating forward and right multiple times, I told the program to space out the squares and draw 5 of them.