Update main.pyx