Supervised Machine Learning Code Example