Henry and William's Python Code