Added docstrings
Added a simple demo file