Coursera python for everybody 5.2 assignment