simplify find_max.py