Better coment
Merge pull request #248 from roydude/v1-train

Better coment