forgot to import numpy