made README example more pythonic