proper gain staging

proper gain staging