A more elegant `argmax` definition.