Updated NumPy notebook