Saddle points exercise | refactored matrix variable to rows