small changes in model