Improve readme
More comments in code