modify candy.py