still a work in progress because the code will only work with the Euler
integration scheme for now.
we need to switch `enforce_unit_orientation` to work on the orientation
itself, and to call it in a different place in the code.