Added docs to model classes