Doc comments for model, slight refactor.