Added training set and ported ipy notebook.