Added some code in the decision models class