Edited readme and docstrings