bug fix and update readme