Exercise #1 of the machine learning class