Fixed bug with returning (-1,-1), added argmax
TODO rewrite to not use argmax for easier to understand code