Revise docs for the model classes