Readme updated, bug fixed