trying to get method working, but it's blocked.
Merge pull request #209 from roydude/v1-train

trying to get method working, but it's blocked.