rewrite setup.py, more simple