Code clean up
Change gains