making some changes in models