Improve docs
Improve readme