Add distance checking while taking treasures