How to control the checkout stages - steps

[geoele]

Hello pros. I use the vm 3  and the medamod to display a 4 difrent modules to every of the 4 checkout steps and i use the pagetype cart, cart.editshipment, and cart.editpayment - e.g.
$vm = JomGenius("virtuemart"); // need this at the start of every rule
if ( $vm->check( "pagetype = cart.editshipment" ) ) return 157;

In the 3 steps everything success. Bat to 4 step means the final confirm page and the thankyou page to, return again the first step module means pagetype cart.

Can any one help with the 2 last steps medamod code ?

Thank you.