Fix W'bal in Train Performance Plot

.. code still needs to be refactored !