Added machine learning code in the IPython file