added missing import numpy