Improved
Added docstring