Improve setup.py
Use taipan library to shorten some code