Updates for 2.0.1 release