Add castle check