Small update to docs and tutorial