Use py_modules in setup.py