A little more pythonic :)