updated README and code file