Same pep8 remark, but I like *your* indentation there