Bump to 2.0.1 to fix some bugs