Update to work with Sublime Text 3/Python 3

Just a little change so it works with Sublime Text 3/Python 3