Optical amplification for communication systems