Update readme
Add comments in the code