fix little bug and update readme