A scheme for polarization control using two laser beams in a non-linear optical medium is studied using both co- and counter-propagating beam geometries. In particular, we show that under certain conditions it is possible for two laser beams to exchange their polarization states. A model accounting for a more realistic, 2D propagation geometry is presented. The 2D model produces drastically different results (compared to the 1D propagation geometry), creating difficulties for implementing polarization control in a realistic setting. A proposal for overcoming these difficulties by reducing the non-linear optical medium to a thin slab is presented.