Updated docs and example