Update to allow pip install