Update Problem2_code