First Machine Learning Homework