add argmax
comment the fields