Update restricted.py