Normalizing model.py