Machine Learning Leetcode Two