Update the answer to problem1 on GitHub