Doc code update for new feature