Update version info for release

Update version info to 4.0.1 for release.