CS536 Machine Learning Project