Update code documentation