Support more Python versions

Fixes #3