some minor changes

I tried to cythonize in setup.py but it kept giving me an error