Updated code in readme example