Add imports in hud.py