Add test database for predicition