Fix bug and update readme