add scenario modelling functionality and improve prob function