Updates setup.py
More tests