Corrected comments.
Merge pull request #142 from roydude/v1-train

Corrected comments.