Code documentation fixes