Simple machine learning algorithm challenge