small changes to models