Move comments from code to docs