Is there any possibility to make the same stage-probability mapping in custom module? 

We have a different types of potentials. And we want to have a different modules with different names, but in each we need this stage-probability mapping. 

Thank you