Fix initial code py2
Merge pull request #3 from cielavenir/patch-2

Fix initial code py2