Added more stuff to setup.py