Fixing example, adding some stuff to docs