Added imports for Numpy