Removed last of test_in in train-function.
Indented comment.