fix and add doc