Some machine learning algorithm code