Added argMax to Math class