Better comment
Merge pull request #272 from roydude/v1-train

Better comment