Can you please run [pep8](https://pypi.python.org/pypi/pep8) on this file and fix the errors?

Why did you use something as complicated as `(i+1)*(i+2)`? Why not just `np.arange(10)` for instance?