Find max element example fix