Change code example in readme