Merge pull request #20 from roydude/v1-train

V1 train
test
Merge pull request #21 from roydude2/v1-train

test
test