machine learning assignment 2