Update version in __init__

Update version to 2.3.1