Exercise: Yield squares with a loop

You can mix yield with a regular for loop. Complete squares(n) so it yields i * i for every i in range(n).
Python
Setting up Python environment...
Output