Update setup.py

0.3.1 version