Add check for numpy