Python Code For Gradient Descent Visualization