Updated setup.py and fixed test.