Small changes in model