Add entry point in setup.py