Add transition matrix visualization