updating NumPy homework