Improve README and docs