cleanup setup.py
add tests