Code documentation improvements