I don't think "diag" is going to help much with the second exercise. Just changing the diagonal doesn't solve this problem. In order to create such an array, one has to know how to remove a whole column from an array. Students only learn this later on in the course.