function g = grad_ex5(x,iex)
% Define gradients for exercise 5

if nargin==1
    iex=1;
end

if iex==1
    g=zeros(2,1);
    g(1)=4*(x(1)-1)+1;
    g(2)=2*(x(2)-2)+1;
elseif iex==2
    g=zeros(2,1);
    g(1)=-400*(x(2)-x(1)^2)*x(1)-2*(1-x(1));
    g(2)=200*(x(2)-x(1)^2);
end