Wrote unit test for _time_steps().
Edited example in docstring of rad_loss_rate() to be a correct working example.