Added docstrings, updated readme