Java- Create the simulator in Java to compare to python