Example to plot gradient descent iterations