Change the PyInit function