Add docstrings to methods and also more tests