Added docstrings and little clean

- Added docstrings