코드리뷰바탕으로 코드 수정 : python code