Added Polarization code