updated README and a bug fix