Almost finished with NumPy excercises