Implemented more Machine Learning