Fix path of the working directory
Merge branch 'master' of https://github.com/zarch/test_ml
Remove unnecessary code