updated code example in readme