Add documentation in doc.go