Project 3: ML model using Python