Add set_dice method