Update to lib/__init__.py