Add comments
Comments, code improvements, README.md fix