added fit diagnostics