bug fix and readme update