The scope of this paper is to analyze the performance is augmented gain EDFA systems using single and multi-wavelength input sources. The performance of an Optical Communication System can be improved by the use of EDFA as an Optical Amplifier. Erbium Doped Fiber Amplifier (EDFA) is an important element in DWDM networks. A significant MATLAB code is developed, which provides the ability to handle multiple channels, thereby allowing to observe the EDFA gain versus wavelength. Since the gain is non-uniform, it is important to know its characteristics in WDM applications where many channels are sent through amplifier. Addition to this includes the forward Amplified Spontaneous Emission (ASE). The resulting model is practical and accurately represents EDFA gain dynamics and forward ASE.