modify the code so that it gives numResCells*waveform_length*numTx*numRx 4D array as output