Updated code and readme.