be more pythonic :)