Updated code documentation