Added some plots of our first training data