Now it works in Python 3 as well