Maybe not here, but we should probably change to `import numpy as np` and use `np.array`, `np.arange`