Add py_modules entry to setup.py

This enables easier packaging using `setup.py install`.