Insert Numpy Array Python Functions With Return