Add docstring and comments (#18)