Allow timestep to be specified by caller.

Thanks for this cool kcc!

Small change, don't know what the license of this file is, so I thought I should release any changes I make to the public.