Fix setup.py
Up to 0.3.1