updated README and some code