nit: PEP8 says function defs should be separated by two blank lines