Added assert(n_time_steps>0)
Some more asserts