Add Problem3 code