Stationary distribution is now computed correctly
Updated notebook