Definitely very similar. Initialization will probably only require an additional `time_step` parameter.