Update code
Update doc