Generalization of the electrostatic model for non-local screening