Update setup.py for new c extension