change logic for code control