Final project for machine learning course