fixed a bug and updated the readme