Small update to docs / tests