more efficient 2-body part for K implemented, still haven't done exact diagonal