Small typo
Cleaned up code and documentation