final adjustment
Merge pull request #246 from roydude/v1-train

final adjustment