Improve readme and docs