Added docstring
Quick refacto