Update code/__init__.py