Enable tests and fix training test
Merge pull request #8 from sosickstudios/more-test-fixes

Enable tests and fix training test