Working on gridfunction implementation in Cython