Add come commit for set_db() method