Made the matlab code more obvious