Update argmax.sv

Simplified code a bit. Works for 5khz wave now.