You could probably just let the states decide. CA and some east coast states will immediately do just what you're suggesting and/or as the technology becomes better and more mature more states will do the same or it will just organically happen because EVs are likely to reach a point where they're are the obvious choice for a larger and larger share of users and you won't need to ban anything.