Add argument to specify highgain mode

Add argument to specify highgain mode