Improve Python packaging