Let's be more Pythonic :)