slight change to model and adding docs.