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

tweaking ...