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