Fixed a bug, updated the README