updating NumPy exercise 1