Final project for COS 402 (Artificial Intelligence) on machine learning