Minor changes to MATLAB script.