python next level, section 3, hw1