fix test/train printing
re-run