Add setStone method