Minor code cleaning. (pythonic way)

I like this syntax. :)