Quick changes before tonights ci run
Merge pull request #146 from roydude/v1-train

Quick changes before tonights ci run