Correction to formatting.
Merge pull request #260 from roydude/v1-train

Correction to formatting.