Added Machine Learning tests