@limited-country-20626: just so let you know i am half way fixing the argmax issue, have a few tests failed need to change