models.py fixed

Hey Mauro sorry I forgot to add some stuff to the one i sent you. Here is the one working.