Simplify Python 2/3 compatibility