Fix for Python 3
Merge pull request #106 from mohd-akram/master

Minor fix for Python 3