Update candy.py