cleaning model.py