Update db/db.go