Updated code and README file.