add setup.py
fix bug