More PEP8 compliance.
Add input parameter.