Update to use machine learning package
Merge pull request #60 from edx/dev

Update to use machine learning package