Only Changed w and gain
Revert "Only Changed w and gain"

This reverts commit 7c835977777a34bf1b98f122e1b54bfbc2c76f27.