added numpy import statmenent...whoops