Finding square root using gradient descent: demo