Python using sum and max - Richest Customer Wealth