Update README

Updated code