Readme updated; minor code update