DOC: Update doctring