Added a gain module