Code cleanup. First part ondiagonal of applyPreconditioner works