adding homework from machine learning class