Add load_db() and test case