Create gradient.py