Doc string cleanup
Add comments
Updated README