fix model.py bugs