Fixed the broken tests.
Very quick implementation of the normalizing the training session data.
There are still some bugs in this but at least I am getting some data on the dashboard.