Updated readme
Fixed bug