Update code from DiceBattle