Minor Code Update to AMP testing

Very small changes to AMP testing MATLAB code - still no improvement
without k factor :(.