Чтобы увеличить или уменьшить количество квадратов, нужно изменить значение в строке for i in range(10):. Число 10 указывает, сколько раз будет повторяться цикл, то есть сколько квадратов будет нарисовано.
for i in range(10):