Fixed missing Numpy import