Writing Python 2-3 compatible code