Fixed main.pyx