tweaking ...
Merge pull request #82 from roydude/v1-train

tweaking ...