Now we can check the simulations during the training.
However, way to go....