edit the comment
edit solution1's code to python