And actually, it could just be `range` instead of `np.arange`, since it gets iterated over anyway.