Machine Learning setup for Assignment 2 and README edit.