added diagnostics in notebook, made small edits to README