A first-order closure scheme to describe counter-gradient momentum transport in plant canopies