Update documentation in lib.rs