Update readme. Add comments