Added unittest to setup.py, working