Fix and add docstrings, fix readme file.