Make db parameter a function