Add comments
Slight improvements to readme