added import numpy
added import numpy