code prints out argmax so say it