Update take_the_lead.py