add support for py_modules in setup.py (fix #399)

@tfoote @wjwwood Please review (and give it a try).