Simulating the Monty Hall problem in python