Give the proper entry_point in setup.py

- Fixes #2.