fix issue #9
Merge pull request #16 from aulphar/master

fix issue #9, also did a pep8 format