Simplify Python 2 / 3 imports a bit by using the six compat module