Update structure of the package for PyPI