Adding in the User model