fixed docstring of the method, added doctsring for test