Update machine_learning.py