Fix up DB query notebook