Began Main Loop, added argmax