Check pointing after implementing general loss/gain for all Ks.
Unit test passes.