Draft implementation of gradient flow complete -- currently using direct forward integration, to be update to Runge-Kutta.