Probably best to move `np.arange(` to the new line.