Some pythonic suggestions
Merge pull request #1 from 1stvamp/patch-1

Some pythonic suggestions