HW 4 added -- machine learning