Python MCQ solution