Add documentation comments (in progress)