updated argmax