models.py update

Left some model code in there by mistake.