Final machine learning project for Stats 202