Updated to 2.0.1 for #8 fix