Homework on machine learning course