Fix README and code