add set_db() method