Hello, nice approach.
In some existing games, rout is contagious to units beside, or units crossing the random run of the rout, and can be stopped by officers galloping back to rally units, or solid unit keeping their files closed.
Prussian brigade were designed to rally crumbling batalions while new ones were put in line.