Forgot to add some fields to the models