Update version to 0.3.1 in preparation for release