It is not a problem of setting the lattice vector in the z(c)-direction. Your setting for that might be good enough.
The problem is that you have to first get an idea of what is the physical meaning of Born charges for 3D and 2D systems, as well as their definition, then go through the BEC_r.py script and modify it according to the definition for 2D systems.