Add treasure position check