Update docs for 2.0.1