Machine learning class project 2.