Update with working DB connection