Updated code in readme