add some code for setup.py