Fix bug and update README.