Fix bugs & setup.py