bug fixed and readme updated