add assign_candy.py