Slight update for a greatest.py (more general approach)