updated readme, fixed a bug