updated README and code