code cleaning
code cleaning + docstring