Minor corrections to model