Added cython wrapper