Added some more graphs to look into different training runs, cleaned some things up