Updated code in README