Tidied up typo
Tidied up comments and docstrings