add load.py and modify models.py