You seem to be on a similar journey to me in many ways. I used to brace and hold when Bertie lost balance (he speeds up and runs through, particularly on the downward transitions). Not quite cracked it but it’s def better. We still need to do circles and shapes to balance him again. My current target is to be able to do a decent canter to trot transition without losing balance for a few strides!