Add/correct documentation