Question 2: A Python solution