made timestep docstrings clearer