Update readme code sample