following the book Machine Learning in Action