PEP8: return statement should be on a new line