Update for Python 3 syntax