Some more idiomatic python

I liked your code very much, it looked pretty good. this makes it a little better to read I think because of less variables
Merge pull request #1 from JustusAdam/patch-1

Some more idiomatic python