Update lecture3(2).py