Support Python 2