section2.1: add castle problem