Updating class docs