[firedrake] Help with implementing steps in simulation