Added some MATLAB code