Fixed missing numpy import