Update setup.py and readme