Touched up code documentation