Turning in NumPy exercise 1