tweak code in read me