Update to v2.0.1