Update extension.py