Add utils.py (forgot to add it earlier)