Added NumPy check code