Made the script Python 3 combatible
Merge pull request #1 from pratyushmittal/master

Made the script Python 3 compatible