creating a gradient descent animation in python 3c4dcd20ca51