Refactor setup.py
Merge pull request #1 from dongweiming/refactor_setup

Refactor setup.py