improve code comments/docstr