Convert the transition matrix to be SxAxS instead of SxSxA

[Lars Lien Ankile]

Having the transition matrix $T$ to be $T: S\times A\times S$ is much more intuitive, as well as consistent with $R$.