Update to use traits in documentation