Part 1: ML Algorithm