Code documentation updated