Very cool. Would love to see this generalized to an interactive visualization of Markov Decision Processes (MDPs).