Added Numpy import to reflect usage in file