Updated setup.py a little bit