https://progiez.com/introduction-to-machine-learning-week-7-assignment-7