Updated readme code snippet