Make it work with both Python 2 and Python 3 by request