Add gradient.pyx to setup.py