added Stan diagnostics