The function find_max should return the largest number in a list.
When you call it, it always returns None. The logic looks right at first glance — but something is fundamentally broken about where the return statement lives.
Fix it without changing the algorithm.