Update README and improve documentation