Fix setup.py
Bump version -> v0.3.1