Added lacking imports to utils.py