Another upgrade would be to randomly reward the desired behavior instead of always rewarding it. This helps to both prevent gaming the system as well as keeping those dopamine spikes high when you unexpectedly get some candy. Using chocolate as a reward is perfect since it provides tryamine to help replenish your dopamine levels.