Added Machine Learning code and datasets