Code for Python 3.x