Add doc comments for model classes