Fix Python 3 support as package

Fixes #5