cleanup

added docstring and removed unnecessary change