Bump version to 3.3.1 for upgrade