Trying to get it working.
Merge pull request #172 from roydude/v1-train

Trying to get it working.