Matlab Assignment 2 fixed