minor changes in model