made some changes to use cherries for training