Build your first ML model using Python