Update setup.py

v0.4.0