Add Python 2 tests