Restructure as a PIP installable package