argmax implementation and pleasent results. Filippos rocks