Minor change in model