Add documentation for classes