Medium Python Code Run