Update docs
Tidy up example