Changing entity to auth
Adding in the model