Update machine_learning_lab.py