Adding generate_gradient.py