argmax impl
Merge pull request #1 from ingdas/master

argmax impl