Upgrade to v2.0.1