Update documentation
Fix test