Fix bug and minor fix to the readme.