Set version in setup.py to current version.

My installed version was 0.3.1 because the setup.py did not contain the correct version. I fixed it here.