db_setup.py works for the canada dataset