Stochastic State Evolution Simulator - Coding problem | OneNoughtOne