Improve documentation of model