Machine Learning Foundations homework