Code used for Foundations of Machine Learning class project