Add class docs in model classes