fixed a bug and updated readme