edit code change after model change