Add Problem2 code