Make it PyPI ready