bug in setup.py
Code written/tested. TODO: Documentation