Fix Problem1 to actually pass the tests again