The >= thing is fine
but what do you mean by use time_step?
I'm using dist_test as a counter, isn't time_step a constant?