MAINT: small refactoring. pep8 compliance, more robust tests
DOC: document the changes to  parameter.