Re-added the User model import