Doc string improved