Improve docstring for formatting