Quick corrections while watching tonight's ci run.
Merge pull request #57 from roydude/v1-train

Quick corrections while watching tonight's ci run.