Need help setting up MATLAB code for this assignment