https://coderprog.com/python-machine-learning-example-2nd/