Nit: PEP8 says 2 blank lines between functions (be consistent with the rest of the code)