Improve comments
Update readme