added probabilistic transitions... not tested yet but hasn't broken the previous thing