A python solution for Monty Hall