Final project for our machine learning class