fixed a bug as well as updated readme